# 导入py5库
import py5

def setup():
    """初始化设置"""
    py5.size(600, 400)  # 设置画布大小为600x400像素
    py5.background(255)  # 设置背景颜色为白色
    py5.no_stroke()  # 不绘制图形轮廓

def draw():
    """每帧绘制的内容"""
    if py5.random(1) < 0.5:  # 随机决定是画矩形还是圆
        py5.fill(py5.random(255), py5.random(255), py5.random(255))  # 设置填充颜色为随机RGB值
        py5.rect(py5.random(py5.width), py5.random(py5.height), 50, 50)  # 在随机位置绘制一个50x50的矩形
    else:
        py5.fill(py5.random(255), py5.random(255), py5.random(255))
        py5.ellipse(py5.random(py5.width), py5.random(py5.height), 30, 30)  # 在随机位置绘制一个直径为30的圆
        # 每秒保存一次图像
        #py5.save_frame("frame-####.png")

# 如果这是主程序（即直接运行这个文件），则启动py5 sketch
if __name__ == '__main__':
    py5.run_sketch()
